"""
python fujian2_groupByCategory_gragh.py
"""

import os
import json
import matplotlib.pyplot as plt
import pandas as pd
from datetime import datetime
from tqdm import tqdm

# 设置文件夹路径
input_path = "../fujian/fujian2/groupByCategory"
output_path = "../fujian/fujian2/groupByCategory/category_plots"

# 创建输出文件夹
os.makedirs(output_path, exist_ok=True)

# 获取所有 JSON 文件
json_files = [f for f in os.listdir(input_path) if f.endswith('.json')]

# 使用 tqdm 显示进度条
for json_file in tqdm(json_files, desc="Processing JSON files"):
    file_path = os.path.join(input_path, json_file)
    
    # 读取 JSON 数据
    with open(file_path, 'r', encoding='utf-8') as f:
        sales_data = json.load(f)
    
    # 将数据转换为 DataFrame
    df = pd.DataFrame(sales_data)
    df['date'] = pd.to_datetime(df['date'])
    df.set_index('date', inplace=True)
    
    # 绘制图表 (散点图)
    plt.figure(figsize=(15, 6))  # 设置长宽比为 2.5:1
    plt.scatter(df.index, df['sales'], color='b', s=10, label="Sales Data")
    plt.title(f'Sales Data for {json_file}')
    plt.xlabel('Date')
    plt.ylabel('Sales')
    plt.legend()
    plt.grid()
    
    # 保存图像
    output_file = os.path.join(output_path, f"{json_file.split('.')[0]}.png")
    plt.savefig(output_file)
    plt.close()

print("所有图像已生成！")
